# -*- coding: utf-8 -*-
# @Time    : 2021/12/17 9:48
# @Author  : xuhan
# @Email   : 1357601817@qq.com
# @File    : get_imgs.py
# @Software: PyCharm
import os
from natsort import natsorted


def get_all_imgs(path):
    # dirpath 代表当前遍历到的目录名
    # dirnames 是列表对象，存放当前dirpath中的所有子目录名
    # filenames 是列表对象，存放当前dirpath中的所有文件名
    filenames_list = []
    for (dirpath, dirnames, filenames) in os.walk(path):
        filenames_list.append(dirpath)
        # filenames = natsorted(filenames)
        filenames_list.append(filenames)
    # print(len(filenames))
    # print(filenames)
    return filenames_list

